lr=2e-5
batch_size=32
epochs=12
pos_weight=25
warmup_steps=0
method=2
time=3
gpu_id=2

CUDA_VISIBLE_DEVICES=$gpu_id python train.py --lr $lr --batch_size $batch_size --epochs $epochs --pos_weight $pos_weight --warmup_steps $warmup_steps --method $method --time $time --init_dense
CUDA_VISIBLE_DEVICES=$gpu_id python test_all_checkpoints.py --lr $lr --batch_size $batch_size --epochs $epochs --pos_weight $pos_weight --warmup_steps $warmup_steps --method $method --time $time --init_dense